Skip to content

Conversation

Johansmm
Copy link
Contributor

@Johansmm Johansmm commented Sep 21, 2021

Updates:

  • load_weight and load_training functions to enable the load weights ".pth", ".ckpt" or by run id.
  • load weights in detr and deformable models

Closes #74

@Johansmm Johansmm added enhancement New feature or request alonet labels Sep 21, 2021
@Johansmm Johansmm self-assigned this Sep 21, 2021
@thibo73800
Copy link
Contributor

load_training can used both run_id & weights with .pth weights as well. But looking at the code it seem to me that load_weights in detr & deformable can't be created based on run_id right ?

@Johansmm
Copy link
Contributor Author

Johansmm commented Sep 22, 2021

@thibo73800 Exactly, because detr is a nn.module and it does not have project_id and run_id. attributes. But instead, I update load_weights function to load weights from .ckpt files. I think that this feature was missing: load weighs from ".ckpt" in a nn.module. What do you thing?

@thibo73800
Copy link
Contributor

Ok, I guess this is not an issue right now. I'm just thinking that at some point it would be great to be able to load models directly from run_id. #85

@thibo73800 thibo73800 merged commit b8cc34a into master Sep 23, 2021
@thibo73800 thibo73800 deleted the 74-load-weights-improve branch September 23, 2021 06:44
@Johansmm Johansmm mentioned this pull request Sep 28, 2021
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
alonet enhancement New feature or request
Projects
None yet
Development

Successfully merging this pull request may close these issues.

load weights from path/run id
2 participants